package com.guanzi.server.util;

import org.apache.poi.hssf.usermodel.*;
/**
 * 导出Excel表的工具类
 * @author guanzi
 * @version 1.0
 * @date 2021/6/15 15
 */
public class Excel {

    /**
     * 导出Excel表方法
     * @param sheetName sheet名称
     * @param title 标题
     * @param values 内容
     * @param wb HSSFWorkbook对象
     * @return
     */
    public static HSSFWorkbook getHSSFWorkbook(String sheetName,String[] title,String[][] values, HSSFWorkbook wb){
        //创建一个HSSFWorkbook，对应一个Excel文件
        if (wb == null){
            wb = new HSSFWorkbook();
        }

        //在workbook中添加一个sheet，对应Excel文件中的sheet
        HSSFSheet sheet = wb.createSheet(sheetName);

        //在sheet中添加表头第0行
        HSSFRow row = sheet.createRow(0);

        //创建单元格，并设置值表头 设置表头居中
        HSSFCellStyle style = wb.createCellStyle();
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER);

        //声明列对象
        HSSFCell cell = null;

        //创建标题
        for (int i = 0; i < title.length; i++){
            //设置length列标题
            cell = row.createCell(i);
            //设置每一列标题的名称
            cell.setCellValue(title[i]);
            //设置每一列标题的样式（居中）
            cell.setCellStyle(style);
        }

        //创建内容
        for (int i = 0; i < values.length; i++){
            //创建行对象
            row = sheet.createRow(i + 1);
            //循环设置每行每列的内容
            for (int j = 0; j < values[i].length; j++){
                row.createCell(j).setCellValue(values[i][j]);
            }
        }

        return wb;
    }
}
